import numpy as np
import matplotlib.pyplot as plt


A=[1,4]
B=[8,9]
X= [A[0],B[0]]   #x坐标点集
Y= [A[1],B[1]]   #y坐标点集

# 等分线段 两点坐标等分（这里是2等分）
# A(Ax,Ay) B(Bx,By) #两点坐标

# X_divide_1 = A[0] + (B[0]-A[0]) /4*1 # x坐标
# Y_divide_1 = A[1] + (B[1]-A[1]) /4*1 # y坐标
# X_divide_2 = A[0] + (B[0]-A[0]) /4*2 # x坐标
# Y_divide_2 = A[1] + (B[1]-A[1]) /4*2 # y坐标
# X_divide_3 = A[0] + (B[0]-A[0]) /4*3# x坐标
# Y_divide_3 = A[1] + (B[1]-A[1]) /4*3 # y坐标

# plt.plot(X,Y) #画线
# plt.plot(X_divide_1, Y_divide_1, 'om')  # 绘制紫红色的圆形的点
# plt.plot(X_divide_2, Y_divide_2, 'om')  # 绘制紫红色的圆形的点
# plt.plot(X_divide_3, Y_divide_3, 'om')  # 绘制紫红色的圆形的点



def Divide_line(A,B,divides):
    '''
    A=[1,4] #A点坐标
    B=[8,9] #B点坐标
    divides 是等分数
    '''
    X=[] # X坐标集
    Y=[] # Y坐标集
    for i in range(1,divides):
        X_divide_i = A[0] + (B[0]-A[0]) /divides*i # x坐标
        Y_divide_i = A[1] + (B[1]-A[1]) /divides*i # y坐标  
        X.append(X_divide_i)
        Y.append(Y_divide_i)
    
    return X, Y

x,y=Divide_line(A,B,6)
print(x)
print(y)
plt.plot(X,Y) #画线
plt.plot(x, y, 'om')  # 绘制紫红色的圆形的点
plt.show()